#pragma once
// 异步日志器的缓冲区

#include <iostream>
#include <vector>
#include <cassert>

#define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)   // 缓冲区初始大小
#define THRESHOLD_BUFFER_SIZE (4 * 1024 * 1024) // 指数增长阈值
#define INCREMENT_BUFFER_SIZE (1 * 1024 * 1024) // 线性增长大小

namespace mylog
{
    class Buffer
    {
    public:
        Buffer()
            : _buffer(DEFAULT_BUFFER_SIZE),
              _writer_idx(0),
              _reader_idx(0)
        {
        }
        // 写数据操作
        void push(const char *data, size_t len)
        {
            // 缓冲区剩余大小不足，可1.扩容 2.返回，由外界阻塞

            // 若缓冲区固定大小，则返回
            // if(len>writeAbleSize()) return;

            // 动态扩容
            ensureEnoughSize(len);

            // 1. 将数据拷贝到缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            // 2. 将写指针偏移
            moveWriter(len);
        }
        // 返回可写大小
        size_t writeAbleSize()
        {
            return (_buffer.size() - _writer_idx);
        }
        // 返回读数据的起始地址
        const char *readBegin()
        {
            return &_buffer[_reader_idx];
        }
        // 返回可读数据的长度
        size_t readAbleSize()
        {
            return _writer_idx - _reader_idx;
        }
        // 移动读指针位置
        void moveReader(size_t len)
        {
            assert(_reader_idx + len <= _writer_idx);
            _reader_idx += len;
        }

        // 重置缓冲区
        void reset()
        {
            _writer_idx = 0;
            _reader_idx = 0;
        }
        // 交换缓冲区
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_writer_idx, buffer._writer_idx);
            std::swap(_reader_idx, buffer._reader_idx);
        }
        // 判断缓冲区是否为空
        bool empty()
        {
            return (_writer_idx == _reader_idx);
        }

    private:
        // 动态扩容
        void ensureEnoughSize(size_t len)
        {
            if (len < writeAbleSize())
                return;
            size_t new_size = 0;
            if (_buffer.size() < THRESHOLD_BUFFER_SIZE)
                new_size = _buffer.size() * 2 + len;
            else
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len;
            _buffer.resize(new_size);
        }
        // 移动写指针位置
        void moveWriter(size_t len)
        {
            _writer_idx += len;
        }

    protected:
        std::vector<char> _buffer; // 缓冲区
        size_t _writer_idx;        // 写指针
        size_t _reader_idx;        // 读指针
    };
}